package com.tevintao.marketingsystem.services;

import android.media.AudioFormat;
import android.media.AudioManager;
import android.media.AudioRecord;
import android.media.AudioTrack;
import android.media.MediaRecorder;
import android.os.Environment;

import com.tevintao.marketingsystem.log.LogManager;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

/**
 * Created by tevintao on 2016/12/8.
 */
public class RecordTest {
    private static RecordTest ourInstance = new RecordTest();

    public static RecordTest getInstance() {
        return ourInstance;
    }

    private RecordTest() {
    }


    int frequency = 16000;
    int channelConfiguration = AudioFormat.CHANNEL_CONFIGURATION_MONO;
    int audioEncoding = AudioFormat.ENCODING_PCM_16BIT;

    public static String path = "/mnt/sdcard/ms/test.pcm";
    private String wavPath = "/ms/voice.wav";
    private boolean isRecording = false;
    private boolean isPlay = false;


    public void selfRecord() {
        File file = new File(Environment.getExternalStorageDirectory().getAbsolutePath() + path);
        // Delete any previous recording.
        if (file.exists())
            file.delete();

        // Create the new file.
        try {
            file.createNewFile();
        } catch (IOException e) {
            throw new IllegalStateException("Failed to create " + file.toString());
        }

        try {
            OutputStream os = new FileOutputStream(file);
            // Create a new AudioRecord object to record the audio.
            int bufferSize = AudioRecord.getMinBufferSize(frequency, channelConfiguration, audioEncoding);
            LogManager.getInstance().logInfo("bufferSize =  " + bufferSize);
            AudioRecord audioRecord = new AudioRecord(MediaRecorder.AudioSource.MIC,
                    frequency, channelConfiguration,
                    audioEncoding, bufferSize);

            byte[] buffer = new byte[bufferSize];
            audioRecord.startRecording();

            isRecording = true;
            while (isRecording) {
                int bufferReadResult = audioRecord.read(buffer, 0, bufferSize);
                LogManager.getInstance().logInfo("bufferReadResult == " + bufferReadResult);
                if (AudioRecord.ERROR_INVALID_OPERATION != bufferReadResult && os != null) {
                    os.write(buffer, 0, bufferReadResult);
                }
            }
            audioRecord.stop();
            os.close();
            LogManager.getInstance().logInfo("record finish");
        } catch (Throwable t) {
            LogManager.getInstance().logError("Recording Failed");
        }
        File outFile = new File(Environment.getExternalStorageDirectory().getAbsolutePath() + wavPath);
        if (outFile.exists())
            outFile.delete();

        try {
            outFile.createNewFile();
        } catch (IOException e) {
            throw new IllegalStateException("Failed to create " + file.toString());
        }
//        copyWaveFile(file.getAbsolutePath(), outFile.getAbsolutePath());
    }


    // 这里得到可播放的音频文件
    private void copyWaveFile(String inFilename, String outFilename) {
        FileInputStream in;
        FileOutputStream out;
        long totalAudioLen;
        long totalDataLen;
        long longSampleRate = frequency;
        int channels = 2;
        long byteRate = 16 * frequency * channels / 8;
        byte[] data = new byte[frequency];
        try {
            in = new FileInputStream(inFilename);
            out = new FileOutputStream(outFilename);
            totalAudioLen = in.getChannel().size();
            totalDataLen = totalAudioLen + 36;
            WriteWaveFileHeader(out, totalAudioLen, totalDataLen,
                    longSampleRate, channels, byteRate);
            while (in.read(data) != -1) {
                out.write(data);
            }
            in.close();
            out.close();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 这里提供一个头信息。插入这些信息就可以得到可以播放的文件。 为我为啥插入这44个字节，这个还真没深入研究，不过你随便打开一个wav
     * 音频的文件，可以发现前面的头文件可以说基本一样哦。每种格式的文件都有 自己特有的头文件。
     */
    private void WriteWaveFileHeader(FileOutputStream out, long totalAudioLen,
                                     long totalDataLen, long longSampleRate, int channels, long byteRate)
            throws IOException {
        byte[] header = new byte[44];
        header[0] = 'R'; // RIFF/WAVE header
        header[1] = 'I';
        header[2] = 'F';
        header[3] = 'F';
        header[4] = (byte) (totalDataLen & 0xff);
        header[5] = (byte) ((totalDataLen >> 8) & 0xff);
        header[6] = (byte) ((totalDataLen >> 16) & 0xff);
        header[7] = (byte) ((totalDataLen >> 24) & 0xff);
        header[8] = 'W';
        header[9] = 'A';
        header[10] = 'V';
        header[11] = 'E';
        header[12] = 'f'; // 'fmt ' chunk
        header[13] = 'm';
        header[14] = 't';
        header[15] = ' ';
        header[16] = 16; // 4 bytes: size of 'fmt ' chunk
        header[17] = 0;
        header[18] = 0;
        header[19] = 0;
        header[20] = 1; // format = 1
        header[21] = 0;
        header[22] = (byte) channels;
        header[23] = 0;
        header[24] = (byte) (longSampleRate & 0xff);
        header[25] = (byte) ((longSampleRate >> 8) & 0xff);
        header[26] = (byte) ((longSampleRate >> 16) & 0xff);
        header[27] = (byte) ((longSampleRate >> 24) & 0xff);
        header[28] = (byte) (byteRate & 0xff);
        header[29] = (byte) ((byteRate >> 8) & 0xff);
        header[30] = (byte) ((byteRate >> 16) & 0xff);
        header[31] = (byte) ((byteRate >> 24) & 0xff);
        header[32] = (byte) (2 * 16 / 8); // block align
        header[33] = 0;
        header[34] = 16; // bits per sample
        header[35] = 0;
        header[36] = 'd';
        header[37] = 'a';
        header[38] = 't';
        header[39] = 'a';
        header[40] = (byte) (totalAudioLen & 0xff);
        header[41] = (byte) ((totalAudioLen >> 8) & 0xff);
        header[42] = (byte) ((totalAudioLen >> 16) & 0xff);
        header[43] = (byte) ((totalAudioLen >> 24) & 0xff);
        out.write(header, 0, 44);
    }


    public void stop() {
        isRecording = false;
    }

    public void selfPlay() {
        isPlay = true;
        File file = new File(path);
        try {
            InputStream is = new FileInputStream(file);

            int minBufSize = AudioTrack.getMinBufferSize(frequency,
                    channelConfiguration,
                    audioEncoding);

            AudioTrack audioTrack = new AudioTrack(AudioManager.STREAM_MUSIC,
                    frequency,
                    channelConfiguration,
                    audioEncoding,
                    (int) (file.length()),
                    AudioTrack.MODE_STREAM);
            audioTrack.play();
            byte[] buffer = new byte[minBufSize * 4];
            int len;
            while (isPlay) {
                len = is.read(buffer);
                if (len <= 0) {
                    LogManager.getInstance().logInfo("play read file finish ");
                    break;
                }
                audioTrack.write(buffer, 0, len);
            }
            LogManager.getInstance().logInfo("play finish ");
            is.close();
            audioTrack.stop();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            LogManager.getInstance().logError("Playback Failed FileNotFoundException");
        } catch (IOException e) {
            e.printStackTrace();
            LogManager.getInstance().logError("Playback Failed IOException");
        }
    }


    public void stopPlay() {
        isPlay = false;
    }


    public void selfPlay(String path, int frequency, int shengdao) {
        isPlay = true;
        File file = new File(path);
        try {
            InputStream is = new FileInputStream(file);

            int minBufSize = AudioTrack.getMinBufferSize(frequency,
                    channelConfiguration,
                    audioEncoding);

            AudioTrack audioTrack = new AudioTrack(AudioManager.STREAM_MUSIC,
                    frequency,
                    shengdao,
                    audioEncoding,
                    10000,
                    AudioTrack.MODE_STREAM);
            audioTrack.play();
            byte[] buffer = new byte[minBufSize * 4];
            int len;
            while (isPlay) {
                len = is.read(buffer);
                if (len <= 0) {
                    LogManager.getInstance().logInfo("play read file finish ");
                    break;
                }
                audioTrack.write(buffer, 0, len);
            }
            LogManager.getInstance().logInfo("play finish ");
            is.close();
            audioTrack.stop();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            LogManager.getInstance().logError("Playback Failed FileNotFoundException");
        } catch (IOException e) {
            e.printStackTrace();
            LogManager.getInstance().logError("Playback Failed IOException");
        }
    }
}
